关于CUDA中的warp shuffle函数说明 |
您所在的位置:网站首页 › GPU warp什么意思 › 关于CUDA中的warp shuffle函数说明 |
以前听过山本前辈【山本A梦我对不起你Orz】细心的讲解。然后忘了。 现在重新理解了一遍、把自己的理解中的要点记了下来 基本的内容(原文):CUDA C PROGRAMMING GUIDE - warp shuffle 函数: https://docs.nvidia.com/cuda/cuda-c-programming-guide/index.html#warp-shuffle-functions
在使用warp shuffle函数时,比如实际用的时候用的是__shfl_up_sync(0xffffffff, res, i),如下面一段计算前缀和(prefix sum)的代码: 1 #include 2 #include 3 #include "cuda_runtime_api.h" 4 #define MASK 0xffffffff 5 6 __global__ void doKernel(int* A, int* B){ 7 ¦ int tid = threadIdx.x; 8 //¦ int val = A[tid]; 9 ¦ int val = 1; 10 ¦ int res = val; 11 ¦ int tmp; 12 ¦ /* 13 ¦ for (int k = 1; k |
今日新闻 |
推荐新闻 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |